package com.gdhengdian.tree_linkedList;

import java.util.HashMap;
import java.util.Map;

/**
 * @author HD-0078 杨炜耀
 * @date 2021-4-7 15:53
 * 单向链表数据结构
 */
public class LinkedList<E> {

    /**
     * 虚拟头节点
     */
    private Node dummyHead;

    /**
     * 链表中元素的个数
     */
    private int size;

    public LinkedList(){
        this.dummyHead = new Node(null, null);
        this.size = 0;
    }

    private void add(int index, E e){
        if(index > size || index < 0){
            throw new IndexOutOfBoundsException("Index: "+index+", Size: "+size);
        }

        Node prev = dummyHead;
        // 移动prev到index前一个节点的位置
        for (int i = 0; i < index; i++) {
            prev = prev.next;
        }

        prev.next = new Node(e, prev.next);
        size++;
    }

    /**
     * 在链表头添加新的元素e
     * @param e 新的元素
     */
    public void addFirst(E e) {
        add(0, e);
    }

    /**
     * 查找链表是否包含e元素
     * @param e 指定元素
     * @return
     */
    public boolean contains(E e){
        Node currentNode = dummyHead.next;
        while (currentNode != null){
            if(currentNode.e.equals(e)){
                return true;
            }
            currentNode = currentNode.next;
        }
        return false;
    }

    /**
     * 从链表删除元素
     * @param e 被删除的元素
     */
    public void removeElement(E e){
        Node prev = dummyHead;
        while (prev.next != null){
            if(prev.next.e.equals(e)){
                break;
            }
            prev = prev.next;
        }
        if(prev.next != null){
            Node delNode = prev.next;
            prev.next = delNode.next;
            delNode.next = null;
            size--;
        }
    }


    /**
     * 获取链表中的元素个数
     * @return 元素个数
     */
    public int getSize() {
        return size;
    }

    /**
     * 链表是否为空
     * @return 为空返回true，否则返回false
     */
    public boolean isEmpty() {
        return size == 0;
    }

    private class Node{
        E e;
        Node next;

        public Node(){
            this(null, null);
        }

        public Node(E e){
            this(e, null);
        }

        public Node(E e, Node next){
            this.e = e;
            this.next = next;
        }

        @Override
        public String toString() {
            return e.toString();
        }
    }
}
